Add test for flags