Move Context to context.py
Remove Context from client.py
import Context to distarray.*
Fix imports to account for moved Context class.