py file for visualization