introducing back the learning rate