Add parameters to modify RMSprop