latest  modifications done in the train, added new functionalities!