Make it compatible with python3