Removing unused import and fixing indentation issue