Add publish status logic