Add ability to set flags