Python calculations for laval nozzle