Fixed one of the tests and added __eq__() to the class