Updated getNoiseMatrix to allow lambda to be a vector