Add a restart option