Context module
Code tidy