I think you need to add __hash__ and __eq__