updating tests, adding migration