Making a witness puzzle

I have been working on this for a while trying to make it work, I have it at the moment so that I can move a square around my area with the arrow keys, but I am unsure how to make the game do it with the mouse instead, any ideas?

To use mouse x and mouse y in lieu of arrows, they need to be input axis instead of input actions. So that mouse x drives the squares left and right, and mouse y drives up and down. Witness seems like they have a maximum move rate so you can just clamp the axis output to some value before running it into an add location node.

So how would I go about changing this, to get the desired outcome,