small tweaks in code using AbTesting