Use generator comprehension, add comment about try/except

Thanks for the suggestions @wasade!