Complete 4th part of assignment