Create challenge2_1.py