clean up example.py