Added feature correlation heatmap function.