Add in ex4.py stub code