added flag for training data..