Sweden’s Sjostrom returns to the pool after elbow surgery