Implement restart button (issue #1)