CD4021BE for Buttons instead of Switches

I'm on my first arduino project, I'm trying to set up a 4x4 grid of LED buttons for a puzzle.

The LEDs I can control using the tutorial found here: https://www.arduino.cc/en/Tutorial/ShiftOut
I used two 74HC595 as suggested in the tutorial, and it worked great. For the shift out, however, the tutorial uses switches instead of buttons. I didn't realize the difference until after I had ordered the CD4021BE as suggested on the page. I'm not familiar enough with microcontrollers to know if it's okay to just swap out the switches with buttons or if that will cause an issue. Switches are constant on/off, but my buttons are momentary. Can I use the CD4021BE at all, or should I just order new microcontrollers made to connect a lot buttons? These are the buttons, for reference: 16mm Illuminated Pushbutton - Blue Momentary : ID 1477 : $1.95 : Adafruit Industries, Unique & fun DIY electronics and kits

Thanks in advance. People who answer others' questions on the internet are the unsung heroes of our age.

(deleted)

It should work well for push buttons if sampled quickly enough.

spycatcher2k:
You only need 8 pins for a 4x4 array of buttons.

There's 16 LEDs as well as 16 buttons. I gotta control them all individually so it's a bit elaborate.

Smajdalf:
It should work well for push buttons if sampled quickly enough.

Sampled quickly enough... the button is a momentary switch, so it'll just send the signal for a very short amount of time. You're saying I can see that quick connection? How do I make sure I'm "sampling quickly enough?"

Try it! Short moment for human is looong time for Arduino.

It seems to work! Thanks. I realized in the end they were talking about momentary switches the whole time... oops lol.