added training parameters