Adds restore method