small changed to registration
updated optimizers