Add Flag tests