Updating to A-B testing code