update code for test data