Add public flag checking