Update unsupervised_learning.py