- Minor changes to training GUI