The swimmer is training for the Olympics