I think it's mostly a UI thing, otherwise the timing is good and the mechanic works well.
Maybe you could the charge bardivided into three equal parts, and as you charge, when a section gets completed it pops out and the gauge changes color as it does now, that way you'd visually see the three punch levels and it'll be clearer which one you is enabled.