messing around to get graph correct