Add auto restart funcion