Make it easier to install the command

Move it to the train directory, so that people don’t have to specify the
output file name.