Added button for restart