Creating a visualizer using python