add html to train.php