Add Optimizer class