Only use a single neuron model and move it to generic module.