more updates to test function