Add python script to draw the graphs