Added unit tests
Merge pull request #9 from Iszak/feature-unit-tests

Added unit tests