WIP manipulate dataset for graph