added trainer class