Add required logging