Add if statement to train_info.html