Thai baby elephant takes a dip in the pool for rehab