Create prob5.py