Create challenge1.py