#24 Restart button implementation