Tidy up db migrate