Ben Stokes returned to training with Durham on Saturday following the nightclub incident that has caused the captain to miss England's second Test against New Zealand.