Hiding custom buttons in XK01/XK02 BADI - IF_EX_VENDOR_ADD_DATA_CS

Hello -

I've recently created a BADI in XK01/XK02 to accomodate putting a custom button on the screen and filling the subsequent screen with radio buttons. I've used IF_EX_VENDOR_ADD_DATA_CS & IF_EX_VENDOR_ADD_DATA. After a succesful implementation the user has come back and asked that the button only be available if the user has entered a Company Code on the XK01/XK02 selection screen.

I've attempted to use the IF_EX_VENDOR_ADD_DATA_{CHECK_ADD_ON_ACTIVE method to deactivate the button if no Company code is entered, however, this method is called before the selection screen is even presented so there's no way to deteremine whether or not they've entered a Company Code. I've been able to use IF_EX_VENDOR_ADD_DATA_CS}SUPPRESS_TAXI_TABSTRIPS to suppress the screen that's called after the button is clicked but it the user would like the button hidden entirely unless a Company Code was provided.

Thanks in advance for any help offered.