add ex6.py