add function for test