Updates to the adjoint TNE2 solver