Improved heatmaps, numpy 1.8.1 fixes

Improved heatmaps to allow only active features to be plotted. Fixed
bugs relating to index slicing in numpy 1.8.1.