Add input handler for moving ball