Add custom restore stream functionality #42

#42