Yeah, I wouldn't mind the number of AT-ATs staying the same. Kinda shows their arrogance, like "as if they'd beat these indestructible machines. Let's just send a few down and get this over with." But adding in another AT-ST here and there could be neat.
And to Shadow, I completely agree about the design element of the PT. It's my favorite part of it, in fact. I don't like it all, but I do like the fact that with each succesive movie, the machines and troopers got closer in design to the OT ones. That's something I looked forward to going into Episode III. And I have a feeling you will definitely see some of your suggestions put into the film.