Add custom restore stream functionality

#42