Added in measurement of sum |Psi|^2