Added restarting after install option