neural network code example