Wrote unit test for heatmap controller