Added a restart button

Self explanatory