1. add test code