Added code for knowledgebase visualizer