Update exer2a_four.py