Changes to the TNE2 solver