heat equation: Nonlinear Function