Added graphics to the Heat Flow lab