add conditional to button