Trump: Sen. McCain returning home to Arizona after treatment