Removed heat map code so that I can make a new branch