Clean up imports and UI.