Update mykmeans.py

clean up code