Updated with advection equation.