you can do it in list comprehension