Visualization bug fixes and overall improvements