Add switch for new difficulty rule