Removed unneeded classes due to better understanding of jgrapht API