augmented -k option in utils.py to account for the 100 percent training set