finite difference method code snippets in python