Create list_comprehension.py