modified code to test.