Updates to the TNE2 source term calculation, fixed bugs