class definition as it was
removed unused imports