Shorter code. Don't handle last NN layer specially.