Add Problem4.py by hand