changes for part2 of the assignment