Add restart option during install