add highlight back in