Small mods..

Signed-off-by: Jouni Korhonen <jounikor@gmail.com>