Add tests for equals method