Heat equation on a square region