Added separate handling of training and demo bots