Add tests for Square and Move classes