add train parameter to constructor