Update solveTransport.m