Added AB test code