minor changes to hw2.py
k means tutorial