just some changes in ex6.py