add more test code
Add benchmark test code