HW4 Modified for Part 1