Create prob6.py