small UI improvements
make all visualize functions private