Added tests

Signed-off-by: Zachary Perlmutter <zrp200@gmail.com>