Create challenge_4.py