Solving Heat Equation In Python Example