Added tests

Addresses #30
Will be fixing them. Do let me know, if anything else needs to be addressed to merge the PR.