Toggle now returns the new state