Add migration and set up database