added example list comprehension
added newlines to inputs so that it looks prettier