Added code for clustering cells