Added graph pruning button + visualization