Fix indentation and minor changes

Signed-off-by: Zlatozar Zhelyazkov <zlatozar@gmail.com>