Updated the learning rate