Added restore message code