Add classes to match choice buttons