Move Context class to its own module, context.py.
Change imports to reflect moving of Context class.
Remove old copy of Context class from client.py
Change __init__.py to reflect move of Context class.