Added Restart button