use comprehension list
simpler loop