Some optimization added.

Signed-off-by: Yuriy Rusinov <yrusinov@gmail.com>