Added ability to turn off custom edge weights