updated import for better python3 compatibility