adding missing class import