Fixed #57. Back button hidden when finished training.