Applied a cleaner approach to the list comprehension example.