stuck in 4wd

So my 5R55E automatic transmission went this summer and I had it rebuilt and put back in (i took it out and put it back in myself). Shifts a little rough into the first two gears but everything else is good EXCEPT I cant tell if my truck is in 4wd or not. I know the electrical components are...