Fix the stemming issue faced in hw6