Adding context to get_context_dict