Added heatmap and finished more or less exercise. Found out how to compute max value of a matrix.