Add visualization of category structure