add functions missing from unit test