Update migration to include new fields