Update code for comment remover, but still buggy.