update - training finished dialog added