Update comprehensions.py