Handle future import of print_function correctly

Previously, it would sometimes get placed above the shebang.
Format