Added code to example.py