Create challenge problem.py