added challenge2.py