Added neural network implementation