Added some graphing