Create challenge2.py