Added unittest
Merge pull request #1 from elbuo8/unittest

Added unittest