Changes:
  1) Changes to stage2 reducer
  2) Added driver code for stage 2