ex5.py done