adding part two of assignment one