* Added a model class for keeping the neural network configuration