Olympic champion Suni Lee back in form after debilitating kidney ailment