write test cases

[gaurav-gogia]

Test cases should help in keeping this repo a bit more professional. Let's write some with filename_test.go and that should help others who might want to see if it works or not :)

[ilmanzo]

if you agree, I'll give a try at this

[gaurav-gogia]

Sure, go for it :)