Updated to include new logging method