For free-form writing and mockups, I extensively use Excalidraw. It isn't feature rich like One Note but works great for me as an infinite blackboard.