added some comments from my thoughts on the train