Addition to neural network classes