Add PyPy compatibility