Pass parameters to optimizer update function.

Pass parameters to optimizer update function. #145