[ADD] Added callback for publish