add learning rate as parameter, and create optimizer