Fixing a bug with updating the graph selection.

New Release - Minor bug fixes