fixed heat map visualization and coding style issues