scatter plot now takes grouping variable