Improve logging.
Add context support.