after some testing, sqrt of diag is necessary. cleaned up a little bit