Add missing test functions