week 5 ex2.py