Add UnitTest
Merge pull request #4 from robert87/master

Add UnitTest