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.