Converted the visualization class to OOP based module.