Added heatmap function and updated readme