I think adding a color (or XBox button icon) to on screen actions would be helpful even for keyboard gameplay
You're probably right, I'm not a fan of the UI, especially for the keyboard. I was thinking of icons representing the action. Might be easier to spot than text
