added neural network code