adding restart option