Change for the architecture search