added some classes and stuffs to the neural network