Simplified code by using a list comprehension
Merge pull request #1 from svisser/patch-1

Simplified code by using a list comprehension