Classes diagram updated for the neural network