Add ability to set log level