Final commit problem3.py