minor changes to plot functionality in castalia script