Updated migration file