Minor changes: Removed unused import and trailing spaces