added rudimentary heatmap coloring