That option is already there. From the FAQ:
Simply click on the Flagstick icon in the upper right hand corner of the game screen at any time to remove and replace the flagstick. When the flagstick has been removed a red slash will be displayed over the icon.