Update class5ex2.py