Create problem001.py