Updated Trainee UI