adding ex6.py