Add migration code to change database schema