Add method to restore previous saved model