WIP Add graphs