Update custom_logging.py