Add restart button, closes #2