Add run2 in main and a small change.

In TinyImage, there was an instantiation inside a for loop that
shouldn't necessarily be there.