adding ex4.py