Added restore functionality to testing module