Some small code tweaks

Signed-off-by: DarioI <dario.incalza@litus.cc>