Small changes and added a main() to VisualizationMain