Adding flags

Sorry for messing it up.