added use directives for necessary classes