moved example code back into example.py