Restart button added