updating code to test