Added new test function