Removed unused import and made a minor whitespace change