Code tidy

Remove unused import and align code.