added A/B test code to original