A feature I missed in OneNote: Handwriting recognition. Both GoodNotes and Notability are able to search your writing, provided it’s somewhat legible.
I'm definitely open for suggestions on the note taking software front. OneNote is sometimes pain to use, but Microsoft have designed it to work relatively seamlessly with the Surface line so it's what I defaulted to. I'll give those a try.