Simple neural network implementation