Added appropriate data migration to schema migration