problem3.py done