Update ex6.py