override func encodeRestorableState(with coder: NSCoder) {

coder.encode(selectedSuperheroId.rawValue, forKey: "imageId")

super.encodeRestorableState(with: coder)

}

This function in the sample app attached saves the id.