Create challenge3_1.py