* First attempt at adding the train command