Add ex6.py