Finite Difference Method In Python