Feat: change flags to square style