Removed visual graph code. Will integrate it on seperate branch.