Improve the scatter plot to be a heat map of the source code