Controller Fix

Updated the button display control logic.