Move Context class to its own module.

This PR makes no API changes, it just moves the `Context` class into its own module, `context.py`. 

One question: Now the `DistArray` class is the only thing in `client.py`, should we change the name of `client.py`?