add function to tests