Added graph stuff