Update diagAlg.R