Trying new version of squaring probability weights.