added missing class import