Add 2 methods, display flag and change button icon