Readd python3 compatibility