Improve neural network example.