you're making a good decision. that train trip was one of the best excursions i have ever taken.
If you are going up on the train and back on the bus, then make sure to sit on the left side of the train as you get on. this is the side which will overlook the edge of the mountain. everyone knows about this trick generally so be sure to get in line earlier so you can board earlier and get the side you want. Have fun!