Add onRestart