Add restore functions
Merge pull request #30 from luanmuniz/restore-original

Add restore functions