update code to test