Add the setup for logging