added new test code