Added tests

Addresses #30
Is this lint?