Fixed code to follow Python conventions regarding equality