Added test for multiple imports in distutils script.

This exposes a bug.
Breaking after finding first import line.

Fixes a bug with multiple import lines. We want to insert ourselves after the
*first* found import line, not the last one...

Also simplified an if/else for when there's no import statement at all.