adding function to test