Add first unit tests
Added tests