Fixed problem with the learning rate