Update sample to include new flags