Small changes on Neural Network: learning rate is added as a parameter