Added better status of training