adding rest of the assignment