Removed unused imports
Small fix for logging