add test of code