Added WB control