Adding functionality to pick the right worker