hw2 Edited with heatmap. Close but not matching the groups