Djokovic returns to training: "I want to play the Australian Open"