Some fixes to visualizer
Use LPU visualizer in demo