Update to work with neural networks