ADD - colors to the scatter plots based on classification