Adding in challenge3.py