fixed std dev output by saving the diagonal std dev instead of the max across all axes, fixes #21