Fixed graph so it works now!