Added testing for K means