Adding shakespeare heatmap