Updated with final list comprehension answers