Moved work function and context argument order.

Also moved the initialisation of these variables to
the `__init__` of `session.py`.