added some graphing