Added future work and fixed some graph cutoff issues