Added python code for plotting glafic output