Create challenge_2.py