Final commit problem4.py