Added additional tests for migration