Added test function