Button now acts as a toggle