Added class for logging