added classifications up to 8 for the neural network