Added restore() method