Where did you get the replacement parts? Can you also get new cogs to replace worn ones?
For most riders, freewheels are inexpensive and the cogs wear out first so it isn't worth the time to rebuild them. The job isn't that difficult as you have discovered but it is a bit painstaking to deal with all of those 1/8" bearings.