[Ankita] Added Visualization Code