Adding python script for plotting
added some nice figs