John Wall and Bradley Beal prepare for Team USA