add more test code