Fixed the plotting and running scripts