Refactored Code - Removed unused import