fixed up example.py so it actually works