Add py_safe function