Fixed the way choose flags, still need to add more to make sure it works smoothly though