Just a suggestion, maybe we can try to use `logger.info()` to replace `print`.