Worked on HW2

Need to finish 2nd analysis and heatmap