Added option to restart setup