changed learning rate to 0.5