COSMIT pep8, removing unused imports
P3K death to the print statement
P3K fix broken doctest and add forgotten print_function import