Add functionality for logging.py