I've had to do the same thing...take it apart four times before I got it working right! Do not re-install it until you get it "bench" working. Its worth your time to sit there and spin the input on the bench, to make sure nothing is binding (especially in 4HI-Street). Really go over it for a half-hour or so...and cycle thru all three modes. What I have found is that, a simple quicky-check yields all is good. But then if you continue to rotate the input...the more the chain and gears will mathematically stray from each other, and you will start to feel binding if its put together incorrectly. If you turn the input once or twice...it feels good. But if you continue to turn it over 20 times...it will bind if something is wrong. Many times I thought I had an issue with 4HI-Lock, and the real issue was that 4HI-Street was bound so tight...I thought I was in 4HI-Lock. So keep at it...play with it before you re-install it. You'll know when its right...everything works smooth as butta...and it will never bind no matter how long you fiddle with it. Don't try to kid yourself by thinking "its just all new and tight and it will loosen up after a few miles". Trust me, if you got issues on the bench...you'll have issues in the truck.
One area I recommend really taking a close look at, is the indexing of the sun gears with the planetary gears. There are marks on the sun gear, and marks on the planetary gears. When you load the planetary gears into the basket...all four hash marks need to be in alignment with the hash marks on the sun gear. The manual shows "dots"...but all the ones I had apart had "lines". Also...the GM manual shows an exploded view of the gear-train, and there are parts actually shown installed "backwards" in the diagram. Good ole GM...I would expect nothing less. If you try to put it together like shown, you'll quickly discover its impossible to assemble it like the diagram. Common sense. Square peg, round hole.
Lastly...make sure the chain is installed in the right direction. It will fit either way...but the teeth are cut to mesh properly in a specific direction. My chains come with a blue link on one side, that tells you which way to assemble it.