finishing off the last bits of the assignment