add testing
test code