Update migration to include data manipulation