Water flow in non-saturated swelling soil