modidied code after get HW5_FEEDBACK