Add first unit test