Added python3 support