Create challenge.py