More optimisations and adding an exponential decay ODE to try plotting and such