The correct way to do this

Adding in __str__ and __repr__