update co-clustering code