finished part 1, working on part 2 of assignment