Added imports for AssessmentControl

Since AssessmentControl was moved to a new package, I added the imports
in the required classes.