added cluster plotting function