Add flag unit testing.