tidy code

Signed-off-by: MikroMan <subic.tine@gmail.com>