Add Set/Get funcs for the context and test