small modification
add comments