Made additions to the assignment