Added a migration for upgrading