Hello,

Still no reply on https://github.com/zk-phi/highlight-stages/issues/10

I prepared a patch and could merge it if you agree.