improvements (minor)

Signed-off-by: Kalman Olah <kalman@inuits.eu>