Add Azkaban2 functionality