After HW4 part1