changed for HW4 part1