add output test
add comment
fix args