Add class for logging