minor updates on visualizations and other bug fixes