added neural network diagram