PEP8 on context.py.

Remove an unused import and tweak some whitespace in `context.py`.