add ex2.py