Add answer to problem 2 of clustering