Add tests

Signed-off-by: Peter Legierski <peter@legierski.net>