Add missing function in init.py