Add database migration code