Added feature matrix size test before calculating heatmap