added small unit test