Adding conditing to should_pivot? method