Python Code For Visualization