Restore function added