Implemented desired restore button functionality