Added restart funcion, changed 'quit' to ask for confirmation