small changes

in neural network