Adding challenge2.py