Modified migration to include index