adding more tests
cleaned up migration