Add ability to override flags + tests