Fix assignment optim.
Update example