Update stokes_driver.m