TODO: more to prune