updated Couette flow solution