Passing tests for list comprehension exercise