fixed exam-2.py