Update code to use CholeskyResult object