HW2 comments 

Sorry! I uploaded this but forgot to make a pull request. 
Format (personal preference) and spelling fixes; comments on the heat map at the end of the code.