add Class Train with init method