Adding ex4.py