Added some graphing code