Create challenge_1.py