update 1.0

added the use of each of the classes in the train class