Add test cases

[bhavin-a]

Hi @Juggernaut9

I think we can add some unit tests as well, using jest. Let me know if it sounds useful, maybe I can work on this. Thanks!

[BhavyaCodes]

Hey im not really familiar with testing, but if you think this is helpful and doesn't conflict or break things then feel free to add!

[bhavin-a]

Cool, let me try to add some basic testing. New things would always be of help :).
Will start on this then as well. Thanks!

[bhavin-a]

Hi @Juggernaut9 , have added a PR #42 for this issue. Thanks.