unit tests
Merge pull request #1 from KsYay/master

unit tests