Add restore class