Fixes in the visualizer frontend UI