Add unit tests
Merge pull request #1 from alicoding/tests

Add unit tests