modifications for hw6