updating a/b test code