Moved graphing out of branch and into own method