make sample code using the list comprehension