Added Visualization Code