Add diagram visuals