HW4 part1 tweaked for submission