Separate scatter plot into categories