Added some structure to the a train example.