Fixed the graphing