add tests code