Fix a bug regrading neuron indecies in __init__