Create problem6.py