Adding tests

Signed-off-by: Borislav Angelov <borislav.angelov@gmail.com>