Fixed log statements to work with colorlog