Quiz on Python List Comprehension