Added challenge2.py