working on part two of the assignment