added tree viewer to show training directories