fixing graph; forgot to add