Transient Heat Conduction 1d Python