Added tests

Addresses #30
Got it!