Work in progress to add legend to the data