Add restart button

Implemented in chrome
Added todo for firefox