Updated logging configuration for more flexibility