Minor refactoring(Optimizer)

Signed-off-by: Sergey Krivohatskiy <s.krivohatskiy@gmail.com>