add learning rate decay