WIP updating neural network diagram