Add close button to trainingController