Adding more changes for migrations