Latest trainer ui changes