Heatmap works, tweaking feature vectors again