Ryan Lochte says he plan to return and thinks Phelps will be back as well