Got list comprehension working to represent the object class