Added unit tests
Merge pull request #2 from Hecktorzr/unit_tests

Unit Tests