add challenge2.py