Complete problem4.py
Complete problem4.py