Incorporated Evan’s comments for neuron layers