Added a restart button