Adds visualization support for diagnostics