augmented -z and -y option in utils.py for 100 percent training set