Some small modifications

Signed-off-by: Itai Koren <itkoren@gmail.com>