Add keyword rendering to UI