Added the getAssignment function