add color print for replace_key.py