fixed heat map figure, added ns