Edited for Part1 HW4