Removed python3 specific import from code