Vishal #0 Fixing visualization