deleted unneccesary import
small refactoring