Adding neuron version of model