improve comment
add a test