update poem_maker.py

list comprehension is Shiva's gift to humanity.