+ added class for neural network