Fixed bug in one_d_heat_diffusion.py