Clean code

Signed-off-by: Andrey Shamis <lolnik@gmail.com>