Finally add the neural network bollocks