add some code for test