Add documentation on context class