New formulation for diffusion

ns * omegas = A * dns_dz + B * ns