Harris County judge returns to work after treatment