Data has to be encapsulated in data set objects, now takes color as an argument