Show wavefunction_scale as a variable in the .plt file

Also has energy addition as a separate addition in the .plt.
Added a few //TODOs and a warning about optimising for speed
Possibly added multithreading but not many functions using this atm