Add data info/output highlighting on editor (slightly broken). Improve graphs, add legends.
Fix ui imports for classes moved to that module