Add restore functions