Added missing function. Fixes #1