Fix the graphing