Change button UI flags to controller driven flags