refactoring code, adding support for visualizing results from different classifiers