slightly more complex neural network