Add test

add the test