Update 6chapter. 4 ex.py