Improve dead code remover