Adding tests

Signed-off-by: Luca Degasperi <luca@lucadegasperi.com>