Add recovery code