minor bug fixes and visualization enhancements