Added a working restart button