small modification
add reducer