Fixing heat equation... again... Lets hope I got it right this time.