wrote python script to plot stuff