Added initial database migration code