improve neuron network builder