Add step to migrate the database