public function updateStatusFlags(&$flags) {

    if (isset($flags['addedtodraft'])) {
        $flags['status_draft'] = '';
    } elseif ($this->stagesDiffer('Stage', 'Live') /*isset($flags['modified'])*/) {
        $flags['status_draft_published'] = '';
    } else {
        $flags['status_published'] = '';
    }
}