Patch 2

fixed HW 4