heat_equation_code.png