Simplify button display logic