Added Objects and edited List Comprehension