Update mykmeans2.py

cleanup, added test case and time comparison