add imports and edit onStop