Add tests for move method