update test code and documentation