Made the gradient in trcov_baseline_grad more obvious.