Graph almost working