Wasn't the guy at the diner watching Tony the same guy who played Dave Scantino's brother-in-law in season 2? Dave Scantino was the friend Tony busted out for his business.

Anyway, I just rewatched it, and I think the biggest thing everyone missed tonight is AJ quoting Tony at the diner after complaining about his work.

AJ: "Remember the good times."
Tony: Oh, you're gonna get smart now?
AJ: No, didn't you say that once? You gotta remember the good times.
Tony: I guess it's true.