adding more test code