Can you add unittests for these api additions?