I usually work on PDF files and using highlight tool to highlight important parts or take notes with pen tool. I can do this with GoodNotes. What I can not do save my work (highlights, notes or other stuff) to my original imported document. I have to export if I want but I think is not a good way, Some other apps can do this, when you work on a document it asks you to reflect work to original document and reflects if you approve.