Tennis: Wawrinka returns to training after knee injury