Add set neuron function