Cleaned up example.py some