ENH: Add learning rate
wip