Create week6-q1.py