Add first unit test. #339