Using list comprehension