Graph Implementation Python