updating tests to pass