wip: Added graph of return data