some minor updates to CustomConjugateGradient