added more test code