Python Code For Navier Stokes Equation