minor change in class import