ref #9: Added unit tests
Added unit tests