Comments update in kMeans algorithm

Hi again @hadi.fathipour-azar could you please check and approve and merge ?

Please check the TODO in particular (and put a clarifying comment therein if the *2 is indeed correct ?)

Thanks !