Add migration for db changes