BUG/REF: python compatibility, clean up imports
REF compat 2to3 example