removed unused imports
replacing print statement with logger