Roger Federer trains with Stefan Edberg