tidy up
Fix learning rate and model selection