Add advection. Not working now