add ex4.py