I agree button controls would be good but it gets a little more complicated when you're near a plug bank with multiple plugs. It's quite hard to guess what the player intends to do upon pressing the action button
1. Pick up entire plug bank
2. Disconnect plug 1
3. Disconnect plug 2
4. Pick up nearby plug (?)
So that's why I went with mouse controls in the end!