Add some tests

Signed-off-by: Derik Evangelista <kirederik@gmail.com>