Add relay restart class