Zlatan has returned to training