Fixed some bugs, added AM and dissipation calculation. Tested MPI vs serial version and the results are statistically indistinguishable