Updating migrations for new schema