Updating log_gen.py