Di Natale back in training