Add challenge5.py