Add a close() method and context-y goodness.
Update example.py