Modified logging setup