Add new unit test

Add new unit test
add one method