Added feature plots for MNIST data