Moved print_function import to the beginning