Add docstring for context module