Fix indentation and minor chnages

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