clean up the visualize
for 3d, vispy does not look that good at the moment, and it's not ready for prime time yet, so it relies on visvis code, but keep it simple