added clustering and figures