Update to fix errors in the graphing