Better lambda calculation

Looks like this scheme works ok. Still, there are numerical
instabilities (?) only with new scheme when SIGMA > 0.5 and dt relatively low.