Added some graph stuff