Kevin Pietersen makes shock England return as specialist white-ball mentor