added more to A/B test function