The next solution to the list comprehension exercise