Add q1.py